from runtasks.chatchat import chat_run
from runtasks.qwen_api import qwen_api
from runtasks.vector_store import search

# 虚拟环境配置 python -m venv .venv
# 安装依赖：pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
if __name__ == "__main__":

    # search()

    # 先运行base_chinese.py生成模型，然后执行命令下载chatglm2-6b-int4模型
    # 下载chatglm2-6b-int4模型命令huggingface-cli download THUDM/chatglm2-6b-int4 --local-dir D:/demo/gitee/python/models/chatglm2-6b-int4 --local-dir-use-symlinks False
    # 然后运行chat_run.py，首次运行会加载model.safetensors文件，需要等待几分钟
    # chat_run()
    
    qwen_api()